package extend

import (
	"fmt"
	"github.com/beego/beego/v2/adapter/orm"
	"github.com/beego/beego/v2/server/web"
	_ "github.com/go-sql-driver/mysql"
	"time"
)

func init() {
	initDb()
}

func initDb() {
	// 设置为 UTC 时间
	orm.DefaultTimeLoc = time.UTC
	// 注册数据库驱动
	orm.RegisterDriver("mysql", orm.DRMySQL)
	sqlConn, _ := web.AppConfig.String("sqlconn")
	// 设置默认的数据库别名，后续使用这个别名操作这个数据库
	err := orm.RegisterDataBase("default", "mysql", sqlConn)
	if err != nil {
		fmt.Println(err.Error())
	}
	// 根据需要可以设置其他ORM参数，如是否显示SQL语句等
	orm.Debug = true
	orm.RunSyncdb("default", false, true)
}
